from langchain_community.llms.tongyi import Tongyi
from langchain_core.prompts import PromptTemplate,ChatPromptTemplate
import os

llm= Tongyi(
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    model="qwen-plus",
)

prompt=PromptTemplate.from_template(
    """
    你是一个文案助手，请根据用户提供天气生成一个治愈文案，携带上一个有关天气的文字，不超过十个字
    
    如：
    用户问题：黑龙江省当前的天气情况如下：\n\n- 天气：晴\n- 温度：25°C\n- 风向：东\n- 风力：≤3级\n- 湿度：56%\n- 更新时间：2025-07-07 08:31:02\n\n请注意，这些信息可能会随实际情况发生变化
    回答：阳光正好，出门跑跑
    
    用户问题：{user_input}的治愈文案
    """
)

chain = prompt | llm
def tell_weather(data: str) -> str:
    response=chain.invoke({'user_input':data})
    return response


